# -*- coding: utf-8 -*-
__author__ = 'bobby'

# 独立使用django的model
import sys
import os

# 获取这个脚本所在的目录，当前是db_tools
pwd = os.path.dirname(os.path.realpath(__file__))
# 把整个项目的根目录，加入到python的根搜索路径之下。相当于sys.path.insert(0, xxx)
# pwd + "../"  找到的是online_store_backend这个项目文件夹，然后加入到sys.path的路径下面
sys.path.append(pwd + "../")
# 应用django的环境变量
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "online_store_backend.settings")

# 启动django，为了使用后面这些django models的东西
import django

django.setup()

from goods.models import GoodsCategory

from db_tools.data.category_data import row_data

for lev1_cat in row_data:
    lev1_intance = GoodsCategory()
    lev1_intance.code = lev1_cat["code"]
    lev1_intance.name = lev1_cat["name"]
    lev1_intance.category_type = 1
    lev1_intance.save()

    for lev2_cat in lev1_cat["sub_categorys"]:
        lev2_intance = GoodsCategory()
        lev2_intance.code = lev2_cat["code"]
        lev2_intance.name = lev2_cat["name"]
        lev2_intance.category_type = 2
        lev2_intance.parent_category = lev1_intance
        lev2_intance.save()

        for lev3_cat in lev2_cat["sub_categorys"]:
            lev3_intance = GoodsCategory()
            lev3_intance.code = lev3_cat["code"]
            lev3_intance.name = lev3_cat["name"]
            lev3_intance.category_type = 3
            lev3_intance.parent_category = lev2_intance
            lev3_intance.save()
